import django_filters

from station.models import Station
from trade.models import Orders


class StationFilter(django_filters.rest_framework.FilterSet):
    """
    站点地址的过滤类
    根据前端地图定位的地址匹配数据库中的该位置的站点
    """
    address = django_filters.CharFilter(field_name="address", lookup_expr="icontains")  # icontains 表示 包含（忽略大小写）

    class Meta:
        model = Station
        fields = ["address"]  # 过滤的字段

class OrdersFilter(django_filters.rest_framework.FilterSet):
    """
    订单地址的过滤类
    根据前端地图定位的地址匹配数据库中的该位置站点的订单
    """
    address = django_filters.CharFilter(field_name="station__address", lookup_expr="icontains")  # icontains 表示 包含（忽略大小写）

    class Meta:
        model = Orders
        fields = ["address"]  # 过滤的字段